from pathlib import Path
import torch

base_path = Path(__file__).parent.parent

CHATLAW_TEXT2VEC_MODEL_PATH = (base_path / "embedding_models" / "ChatLaw-Text2Vec").__str__()
M3E_BASE_MODEL_PATH = (base_path / "embedding_models"/ "m3e-base").__str__()
BGE_BASE_MODEL_PATH = (base_path / "embedding_models"/ "bge-base").__str__()
TEXT2VEC_LARGE_MODEL_PATH = (base_path / "embedding_models"/ "text2vec_large").__str__()
MULTILINGUAL_SMALL_MODEL_PATH = (base_path / "embedding_models"/ "multilingual-e5-small").__str__()

RERANK_MODEL_PATH = (base_path / "rerank_model"/ "bge-reranker-base").__str__()

device = torch.device("cuda:1")
